LightGBM
LightGBMClassifier
- Python
- Scala
from synapse.ml.lightgbm import *
lgbmClassifier = (LightGBMClassifier()
.setFeaturesCol("features")
.setRawPredictionCol("rawPrediction")
.setDefaultListenPort(12402)
.setNumLeaves(5)
.setNumIterations(10)
.setObjective("binary")
.setLabelCol("labels")
.setLeafPredictionCol("leafPrediction")
.setFeaturesShapCol("featuresShap"))
import com.microsoft.azure.synapse.ml.lightgbm._
val lgbmClassifier = (new LightGBMClassifier()
.setFeaturesCol("features")
.setRawPredictionCol("rawPrediction")
.setDefaultListenPort(12402)
.setNumLeaves(5)
.setNumIterations(10)
.setObjective("binary")
.setLabelCol("labels")
.setLeafPredictionCol("leafPrediction")
.setFeaturesShapCol("featuresShap"))
Python API: LightGBMClassifier | Scala API: LightGBMClassifier | Source: LightGBMClassifier |
LightGBMRanker
- Python
- Scala
from synapse.ml.lightgbm import *
lgbmRanker = (LightGBMRanker()
.setLabelCol("labels")
.setFeaturesCol("features")
.setGroupCol("query")
.setDefaultListenPort(12402)
.setRepartitionByGroupingColumn(False)
.setNumLeaves(5)
.setNumIterations(10))
import com.microsoft.azure.synapse.ml.lightgbm._
val lgbmRanker = (new LightGBMRanker()
.setLabelCol("labels")
.setFeaturesCol("features")
.setGroupCol("query")
.setDefaultListenPort(12402)
.setRepartitionByGroupingColumn(false)
.setNumLeaves(5)
.setNumIterations(10))
Python API: LightGBMRanker | Scala API: LightGBMRanker | Source: LightGBMRanker |
LightGBMRegressor
- Python
- Scala
from synapse.ml.lightgbm import *
lgbmRegressor = (LightGBMRegressor()
.setLabelCol("labels")
.setFeaturesCol("features")
.setDefaultListenPort(12402)
.setNumLeaves(5)
.setNumIterations(10))
import com.microsoft.azure.synapse.ml.lightgbm._
val lgbmRegressor = (new LightGBMRegressor()
.setLabelCol("labels")
.setFeaturesCol("features")
.setDefaultListenPort(12402)
.setNumLeaves(5)
.setNumIterations(10))
Python API: LightGBMRegressor | Scala API: LightGBMRegressor | Source: LightGBMRegressor |